Fixed a small whoops