Fixed issue thanks to jake