FIX: Fix docstring