Fixed small problem in header