Fixed an issue that was caused by a forgetten line