Small task fix