Small corrections for compiling