Fixed problem with error