Fixed minor bug caught by CI