Fixed a small oversight