Add nil-handling