FIX: Fix simple test