Add checks for if things might go wrong