Fixed forgot to stage my fixes