Fixed a stupid forgotten method bug