fixed something small in the cod