Small fixes (unused header etc.)
Fixed a small mistake.