Fixed another problem