Added a small test