Fixed a small problem