FIX: Add simple test