Fix uncaught edge case