Fixed problem code