Add world check for walls