Minor fix when width and height are not the same.
Minor fix when width and height are not the same.

tambetm/simple_dqn,tambetm/simple_dqn,loofahcus/simple_dqn,tambetm/simple_dqn,loofahcus/simple_dqn,loofahcus/simple_dqn