Fixed a remaining mistake